from efficientnet.efficientnet_utils import efficientnet_b0, efficientnet_b1,efficientnet_b2,efficientnet_b3,efficientnet_b4
from SWIN_VIT.model import swin_tiny_patch4_window7_224,swin_small_patch4_window7_224,swin_base_patch4_window7_224
from VIT.models_vit import vit_large_patch16, vit_base_patch16
from timm.models import create_model
model_dict = {
    "efficientnetb0":efficientnet_b0,
    "efficientnetb1":efficientnet_b1,
    "efficientnetb2":efficientnet_b2,
    "efficientnetb3":efficientnet_b3,
    "efficientnetb4":efficientnet_b4,
    "SWIN_VIT_tiny_patch4_window7_224":swin_tiny_patch4_window7_224,
    "SWIN_VIT_small_patch4_window7_224":swin_small_patch4_window7_224,
    "SWIN_VIT_base_patch4_window7_224":swin_base_patch4_window7_224,
    "vit_large_patch16":vit_large_patch16,
    "vit_base_patch16":vit_base_patch16,
    "resnet50":create_model,
    "convnext_tiny_hnf":create_model,
    "efficientnetv2_rw_t":create_model
    }
